package tech.aistar.wyymusicplayer.util;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import tech.aistar.wyymusicplayer.pojo.User;
import java.util.Date;

public class JWTUtils {

    // 过期时间（一周）
    private static final long EXPIRE = 7 * 24 * 60 * 60 * 1000;

    // 秘钥（建议从配置文件读取）
    private static final String SECRET = "success";

    // 令牌前缀
    private static final String TOKEN_PREFIX = "aistar";

    // 令牌主题
    private static final String SUBJECT = "login";

    /**
     * 生成 JWT Token
     */
    public static String geneJsonWebToken(User user) {
        String token = Jwts.builder()
                .setSubject(SUBJECT)
                .claim("id", user.getId())       // 用户ID
                .claim("username", user.getUsername())
                .setIssuedAt(new Date())
                .setExpiration(new Date(System.currentTimeMillis() + EXPIRE))
                .signWith(SignatureAlgorithm.HS256, SECRET)
                .compact();
        return TOKEN_PREFIX + token; // 添加前缀
    }

    /**
     * 验证 Token 并返回用户ID
     */
    public static String validateTokenAndGetUserId(String token) {
        try {
            Claims claims = checkJWT(token);
            return (claims != null) ? claims.get("id").toString() : null;
        } catch (Exception e) {
            return null;
        }
    }

    public static Claims parseJWT(String token) {
        return Jwts.parser()
                // 修改签名密钥的使用方式
                .setSigningKey(SECRET) // 使用统一的密钥变量
                .parseClaimsJws(token.replace(TOKEN_PREFIX, "")) // 添加token前缀处理
                .getBody();
    }

    /**
     * 通用 Token 校验方法
     */
    public static Claims checkJWT(String token) {
        try {
            return Jwts.parser()
                    .setSigningKey(SECRET)
                    .parseClaimsJws(token.replace(TOKEN_PREFIX, "")) // 移除前缀
                    .getBody();
        } catch (Exception e) {
            return null;
        }
    }
}